class Config:
    def __init__(self):
        self.root_path = r'F:\GroupFive\TouManFen'
        # 原始数据路径
        self.train_datapath = self.root_path + r'\01-data\train.csv'
        self.test_datapath = self.root_path + r'\01-data\test.csv'
        self.dev_datapath = self.root_path + r'\01-data\dev.csv'
        # 类别文档
        self.class_doc_path = self.root_path + r"\01-data\class.json"
        # 停用词路径
        self.stop_words_path = self.root_path + r"\01-data\stopwords.txt"

        # 处理后的数据路径
        self.process_train_datapath = self.root_path + r"\02-rf\final_data\train_process.csv"
        self.process_test_datapath = self.root_path + r"\02-rf\final_data\test_process.csv"
        self.process_dev_datapath = self.root_path + r"\02-rf\final_data\dev_process.csv"

        # 保存模型路径
        self.rf_model_save_path = self.root_path + r"\02-rf\save_model\rf_model.pkl"
        self.tfidf_model_save_path = self.root_path + r"\02-rf\save_model\tfidf_model.pkl"
        # 模型预测结果
        self.model_predict_result = self.root_path + r"\02-rf\result\predict_result.txt"


if __name__ == '__main__':
    config = Config()
    print(config.train_datapath)